# coding: utf-8

from werkzeug.routing import BaseConverter
from flask import session, jsonify, g
from app.utils.response_code import RET
import functools

class ReConverter(BaseConverter):
    """"""
    def __init__(self,url_map,regex):
        # 调用父类的初始化方法
        super(ReConverter,self).__init__(url_map)
        # 保存正则表达式
        self.regex = regex

def login_required(view_func):

    @functools.wraps(view_func)
    # 判断用户的登录状态
    def wrapper(*args, **kwargs):
        user_id = session.get("user_id")

        # 如果用户是登录的，执行视图函数
        if user_id is not None:
            # 将user_id 保存到 g 对象中，在视图函数中可以通过g对象获取保存数据
            g.user_id = user_id
            return view_func(*args, **kwargs)
        else:
            # 如果未登录，返回登录的信息
            return jsonify(code = RET.SESSIONERR, msg = "用户未登录")
    return wrapper

# @login_required
# def set_user_avatar():
#     user_id = g.user_id
#     return